package com.ahpu.lyf.sparsearray;

import java.util.Arrays;
import java.util.Random;
import java.util.function.Predicate;

public class SpraseArray {
    public static void main(String[] args) {
        int[][] array = getArray(5, 5);
        printArray("生成的二维数组",array);
        int[][] sparesArray = toSparesArray(array);
        printArray("转换为稀疏数组",sparesArray);
        printArray("转换回二维数组",parseSparseArr(sparesArray));

    }

    /**
     * 将一个二维数组转换为稀疏数组，采用
     * @param arr 传入的二维数组
     */
    public static int[][] toSparesArray(int[][] arr){
//        匿名对象
//        var obj = new Object() {
//            int count = 0;
//        };
//        Arrays.stream(arr).forEach(x-> Arrays.stream(x).forEach(y->{
//            if (y!=0)
//                obj.count +=1;
//        }));

        int count=0;
        for (int i = 0; i < arr.length; i++) {
            for (int j = 0; j < arr[0].length; j++) {
                if (arr[i][j]!=0)
                    count++;
            }
        }

//        int[][] sparseArr=new int[obj.count+1][3];
        int[][] sparseArr=new int[count+1][3];
        sparseArr[0][0]=arr.length;
        sparseArr[0][1]=arr[0].length;
//        sparseArr[0][2]=obj.count;
        sparseArr[0][2]=count;
        int index=1;
        for (int i = 0; i < arr.length; i++) {
            for (int j = 0; j < arr[0].length; j++) {
                if (arr[i][j]!=0){
                    sparseArr[index][0]=i;
                    sparseArr[index][1]=j;
                    sparseArr[index][2]=arr[i][j];
                    index++;
                }
            }
        }
        return sparseArr;
    }

    /**
     * 将稀疏数组转换为二维数组
     * @param sparseArr 要转换的稀疏数组
     * @return 转换后的二维数组
     */
    public static int[][] parseSparseArr(int[][] sparseArr){
        int[][] arr=new int[sparseArr[0][0]][sparseArr[0][1]];
        Arrays.stream(sparseArr).filter(new Predicate<int[]>() {
            @Override
            public boolean test(int[] ints) {
                return sparseArr[0] != ints;
            }
        }).forEach(x->{
            arr[x[0]][x[1]]=x[2];
        });
        return arr;
    }

    /**
     * 返回一个指定的二维数组
     * @param x 二维数组行数
     * @param y 二维数组列数
     * @return 返回指定二维数组
     */
    public static int[][] getArray(int x,int y){
        Random random=new Random();
        int num =Math.min(x,y);
        int[][] arr=new int[x][y];
        int count=random.nextInt(x*y);
        for (int i = 0; i < count; i++) {
            arr[random.nextInt(num)][random.nextInt(num)]= random.nextInt(100);
        }
        return arr;
    }

    /**
     * 打印二维数组
     * @param arr 传入二维数组
     */
    public static void printArray(String message,int[][] arr){
        System.out.println("message ： " + message);
        Arrays.stream(arr).forEach(x->{
            Arrays.stream(x).forEach(y-> System.out.print(y+"\t"));
            System.out.println("");
        });
    }
}
